package com.example.lotterysystem.common.utils;

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;
import org.springframework.util.CollectionUtils;

import java.util.Collection;
import java.util.concurrent.TimeUnit;

@Slf4j
@Component
public class RedisUtil {

    @Autowired
    private StringRedisTemplate stringRedisTemplate;

    /**
     * 设置超时时间
     * @param key
     * @param value
     * @param ttl
     * @param timeUnit 超时时间单位
     * @return
     */
    public boolean set(String key, String value, Long ttl, TimeUnit timeUnit) {
        try {
            if (null != key && ttl > 0) {
                stringRedisTemplate.opsForValue().set(key, value, ttl, timeUnit);
                return true;
            }
            return false;
        } catch (Exception e) {
            log.error("RedisUtil set({}, {}, {}, {})", key, value, ttl, timeUnit, e);
            return false;
        }
    }

    /**
     * 设置超时时间，单位为 秒
     * @param key
     * @param value
     * @param ttl 单位为 秒
     * @return
     */
    public boolean set(String key, String value, Long ttl) {
        return set(key, value, ttl, TimeUnit.SECONDS);
    }

    /**
     * 不设置超时时间
     * @param key
     * @param value
     * @return
     */
    public boolean set(String key, String value) {
        try {
            if (null != key) {
                stringRedisTemplate.opsForValue().set(key, value);
                return true;
            }
            return false;
        } catch (Exception e) {
            log.error("RedisUtil set({}, {})", key, value, e);
            return false;
        }
    }

    /**
     * 获取 key 对应 value
     * @param key
     * @return
     */
    public String get(String key) {
        try {
            if (null != key) {
                return stringRedisTemplate.opsForValue().get(key);
            }
            return null;
        } catch (Exception e) {
            log.error("RedisUtil get({})", key, e);
            return null;
        }
    }

    /**
     * 判断 key 是否存在
     * @param key
     * @return
     */
    public boolean hasKey(String key) {
        try {
            return stringRedisTemplate.hasKey(key);
        } catch (Exception e) {
            log.error("RedisUtil hasKey({})", key, e);
            return false;
        }
    }

    /**
     * 为 key 设置超时时间，单位为 timeUnit
     * @param key
     * @param ttl
     * @param timeUnit
     * @return
     */
    public boolean setExpire(String key, long ttl, TimeUnit timeUnit) {
        try {
            if (null != key && ttl > 0) {
                return stringRedisTemplate.expire(key, ttl, timeUnit);
            }
            return false;
        } catch (Exception e) {
            log.error("RedisUtil setExpire({}, {}, {})", key, ttl, timeUnit);
            return false;
        }
    }

    /**
     * 设置超时时间，单位为 秒
     * @param key
     * @param ttl
     * @return
     */
    public boolean setExpire(String key, long ttl) {
        return setExpire(key, ttl, TimeUnit.SECONDS);
    }

    /**
     * 获取 key 的超时时间
     * 获取失败时返回 -1
     * @param key
     * @return
     */
    public long getExpire(String key) {
        try {
            if (null != key) {
                return stringRedisTemplate.getExpire(key);
            }
            return -1L;
        } catch (Exception e) {
            log.error("RedisUtil getExpire({})", key);
            return -1L;
        }
    }

    /**
     * 删除 key，删除成功，返回删除个数，删除失败，返回 -1
     * @param key
     * @return
     */
    public long del(String... key) {
        try {
            if (null != key && key.length > 0) {
                if (key.length == 1) {
                    return stringRedisTemplate.delete(key[0]) ? 1: 0;
                } else {
                    return stringRedisTemplate.delete((Collection<String>) CollectionUtils.arrayToList(key));
                }
            }
            return -1L;
        } catch (Exception e) {
            log.error("RedisUtil del({})", key);
            return -1L;
        }
    }
}
